#!/usr/bin/env python
# Copyright (c) Facebook, Inc. and its affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

from distutils.core import setup, Extension as _Extension
from distutils.command.build_ext import build_ext as _build_ext
from distutils.errors import CCompilerError, DistutilsError, CompileError

import sys


class Extension(_Extension):
    """Add an `optional` kwarg, to allow skipping on failure"""
    def __init__(self, name, sources, optional=False, **kwargs):
        _Extension.__init__(self, name, sources, **kwargs)
        self.optional = optional


class build_ext(_build_ext):
    """Build extensions, but skip on failure if optional is set"""
    def build_extensions(self):
        self.check_extensions_list(self.extensions)

        for ext in self.extensions:
            try:
                self.build_extension(ext)
            except (CCompilerError, DistutilsError, CompileError) as e:
                if not ext.optional:
                    raise
                self.warn('building extension "%s" failed: %s' %
                          (ext.name, e))


fastprotomod = Extension(
    'thrift.protocol.fastproto',
    sources = ['protocol/fastproto.cpp'],
    libraries=['thriftcpp2', 'thriftprotocol', 'folly'],
    extra_compile_args=['-std=c++1y'],
    optional=True,
)

version_info = sys.version_info
boost_python = "boost_python-py{}{}".format(version_info[0], version_info[1])

cppservermod = Extension(
    'thrift.server.CppServerWrapper',
    sources=['server/CppServerWrapper.cpp'],
    include_dirs=['../../../'],
    libraries=[boost_python, 'thriftcpp2', 'folly', 'wangle'],
    extra_compile_args=['-std=c++1y', '-fno-strict-aliasing'],
    optional=True,
)

setup(name = 'Thrift',
      version = '0.1',
      description = 'Thrift Python Libraries',
      author = 'Thrift Developers',
      author_email = 'thrift-dev@incubator.apache.org',
      url = 'http://incubator.apache.org/thrift/',
      license = 'Apache License 2.0',
      packages = [
        'thrift',
        'thrift.protocol',
        'thrift.transport',
        'thrift.server',
        'thrift.util',
      ],
      package_dir = {'thrift' : '.'},
      ext_modules = [fastprotomod, cppservermod],
      cmdclass={'build_ext': build_ext},
      )
